
#include <torch/extension.h>
#include "rasterize_points.h"

PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
  m.def("rasterize_points", &RasterizePointsCUDA);
  m.def("rasterize_points_backward", &RasterizePointsBackwardCUDA);
  m.def("mark_visible", &markVisible);
}